"""使用Pandas对Excel使用分组与聚合对员工业绩进行汇总"""
"""汇总各柜台的销售总额；统计每个员工交易额平均值；计算每个员工交易额的中值"""
import pandas as pd

# 设置列对齐
pd.set_option('display.unicode.ambiguous_as_wide', True)
pd.set_option('display.unicode.east_asian_width', True)

# 读取全部数据，使用默认索引
df = pd.read_excel(r'resources\超市营业额2.xlsx')

# 汇总各柜台的销售总额
print('各柜台的销售总额'.center(20, '='))
# 单列作为分组字段，设置索引
print(df.groupby('柜台')['交易额'].sum(), end='\n\n')
# 单列作为分组字段，不设置索引
# print(df.groupby(by='柜台', as_index=False).sum(), end='\n\n')

# 统计每个员工交易额平均值，并保留两位小数
print('每个员工交易额平均值'.center(20, '='))
print(df.groupby('工号')['交易额'].mean().round(2), end='\n\n')

# 计算每个员工交易额的中值
print('每个员工交易额的中值'.center(20, '='))
print(df.groupby('工号')['交易额'].median().round(2), end='\n\n')
